# _*_ coding : utf-8  _*_
# @Time : 2025/3/10 00:14
# @Author : 马梓航
# @File : vit_train
# @Project : VIT（VIT各种版本实现+小马实现版
from vit_framework import thisisvit as creat_model

import torch
from torch import nn
import torchvision

vit=creat_model()

#vit模型最好使用预训练权重，要不然很麻烦